from mirai import Mirai
from robot_core import web_socket_helper


class RobotFactory:

    # TODO 从配置文件中读取qq号
    # TODO 添加需要服务的群
    @staticmethod
    def __get_robot_id(robot_name):
        return 1197884312

    def get_robot(self, robot_name):
        bot = Mirai(
            qq=self.__get_robot_id(robot_name),
            adapter=web_socket_helper.get_adapter()
        )

        return bot
